import os


def f():
    base_pth = "./trained_model/Douban/GNN_based"
    embed_dim_list = [16, 32, 64]
    lr_list = [0.1, 0.01, 0.001]
    embedding_type_list = ["TransE", "TransR"]
    laplacian_type_list = ["symmetric", "random-walk"]
    aggregation_type_list = ["gcn", "graphsage", "lightgcn"]
    layers_list = [2, 4, 8]
    for i0, dim in enumerate(embed_dim_list):
        for i1, lr in enumerate(lr_list):
            for i2, em_type in enumerate(embedding_type_list):
                for i3, lap_type in enumerate(laplacian_type_list):
                    for i4, agg_type in enumerate(aggregation_type_list):
                        for i5, nlayers in enumerate(layers_list):
                            pth = os.path.join(
                                base_pth,
                                f"nlayer{nlayers}_dim{dim}_lr{lr}_{lap_type}_{em_type}_{agg_type}",
                                "metrics0.tsv",
                            )
                            try:
                                open(pth)
                            except:
                                pth=pth.replace('/metrics0.tsv','')
                                # os.system(f'rm -rf {pth}')
                                print(pth)


if __name__ == "__main__":
    f()
